const { prisma } = require('../config/prisma');

async function addTestChatCredits() {
  try {
    console.log('开始为测试用户添加聊天消息额度...');

    // 为所有用户添加一些初始聊天消息额度
    const result = await prisma.users.updateMany({
      data: {
        chat_credits: 5 // 给每个用户5条免费聊天消息
      }
    });

    console.log(`成功为 ${result.count} 个用户添加了5条聊天消息额度`);
  } catch (error) {
    console.error('添加聊天消息额度失败:', error);
  } finally {
    await prisma.$disconnect();
  }
}

// 如果直接运行此脚本
if (require.main === module) {
  addTestChatCredits();
}

module.exports = addTestChatCredits;
